from keras import Input
from keras.layers import Flatten

from .base_layer import BaseLayer
from config import ModelLayers
from entity.model_describe import ModelDescribe


class FlattenLayer(BaseLayer):

    def transfer(self, model_list_line: list, inputs: Input, models: list = None):
        return Flatten()(inputs)
if __name__ == "__main__":
    m = FlattenLayer().persistence([0],1)
    input()